<?php
/**
 * @title 726. 原子的数量 - 递归+栈
 * @author start2004
 */

// ini_set("memory_limit", "128M");

// class Solution {
//
//     function find() {
//
//     }
// }

class Solution {

    /**
     * @param String $formula
     * @return String
     */
    function countOfAtoms($formula) {
        /**
         * @since 2021-07-05 统计元素
         */
        $cntArray = self::find($formula, 0, strlen($formula));

        /**
         * @since 2021-07-05 哈希
         */
        $hashArray = $elementArray = [];
        foreach ($cntArray as $e => $n){
            $elementArray[] = $e;
            $hashArray[$e] = $n;
        }

        /**
         * @since 2021-07-05 排序
         */
        sort($elementArray);

        /**
         * @since 2021-07-05 返回
         */
        foreach ($elementArray as $k => $e){
            $n = $hashArray[$e];
            if($n >= 2){
                $elementArray[$k] .= "{$n}";
            } else {}
        }

        /**
         * @return
         */
        return implode("", $elementArray);
    }

    /**
     * @title 统计
     * @author start2004
     * @since 2021-07-05 Created
     *
     * @param string $f
     * @param int $s
     * @param int $e
     * @param int $m
     * @return array[String => int]
     */
    function find($f = '', $s = 0, $e = 0, $m = 1){
        /**
         * @since 2021-07-05 括号数量
         */
        $n = 0;

        /**
         * @since 2021-07-05 统计
         */
        $cntArray = [];

        /**
         * @since 2021-07-05 队列
         */
        $stackArray = [];
        for($i=$s; $i<$e; $i++){
            $c = $f[$i];

            if($c == "("){
                $stackArray[] = $i+1;
            } elseif($c == ")") {
                $n = array_pop($stackArray);
                if(count($stackArray) == 0){
                    /**
                     * @since 2021-07-05 括号后是否数字, 求数字
                     */
                    $num = "";
                    for($j=$i+1; $j<$e; $j++){
                        if(is_numeric($f[$j])){
                            $num .= $f[$j];
                        } else {
                            break;
                        }
                    }

                    /**
                     * @since 2021-07-05 数字格式化
                     */
                    if($num){
                        $num = intval($num)*$m;
                    } else {
                        $num = $m;
                    }

                    /**
                     * @since 2021-07-05 递归
                     */
                    $cntArray = self::mergeCnt($cntArray, self::find($f, $n, $i, $num));
                    $i = $j-1;
                } else {}
            } else {
                /**
                 * @since 2021-07-05 内部小括号不处理
                 */
                if(count($stackArray) == 0){
                    $element = $f[$i];

                    /**
                     * @since 2021-07-05 后续小写字母带进来
                     */
                    for($j=$i+1; $j<$e; $j++){
                        if(strpos("abcdefghijklmnopqrstuvwxyz", $f[$j]) !== false){
                            $element .= $f[$j];
                        } else {
                            break;
                        }
                    }

                    /**
                     * @since 2021-07-05 数字
                     */
                    $num = "";
                    for(; $j<$e; $j++){
                        if(is_numeric($f[$j])){
                            $num .= $f[$j];
                        } else {
                            break;
                        }
                    }
                    $i = $j-1;

                    /**
                     * @since 2021-07-05 数字格式化
                     */
                    if($num){
                        $num = intval($num)*$m;
                    } else {
                        $num = $m;
                    }

                    /**
                     * @since 2021-07-05 记录元素统计
                     */
                    if(isset($cntArray[$element])){
                        $cntArray[$element] += $num;
                    } else {
                        $cntArray[$element] = $num;
                    }
                } else {}
            }
        }

        /**
         * @return
         */
        return $cntArray;
    }

    /**
     * @title 合并统计数组
     * @author start2004
     * @since 2021-07-05 Created
     *
     * @param array $aArray
     * @param array $bArray
     * @return array
     */
    function mergeCnt($aArray = [], $bArray = []){
        /**
         * @since 2021-07-05 遍历
         */
        foreach ($bArray as $k=>$v){
            if(isset($aArray[$k])){
                $aArray[$k] += $v;
            } else {
                $aArray[$k] = $v;
            }
        }

        /**
         * @return
         */
        return $aArray;
    }
}

/**
 * @url http://127.0.0.1/leetcode/202107/2021.07.05.php
 */
$datas = [
    // ["H2O"], // H2O
    // ["Mg(OH)2"], // H2MgO2
    // ["K4(ON(SO3)2)2"], // K4N2O14S4
    // ["Be32"], // Be32
    // ["H11He49NO35B7N46Li20"], // B7H11He49Li20N47O35
    ["(N13O9Be)37(LiC50B35)38(Li33HHBe14He5ON50N)27(H3C)2He14C34Li33C33He15N14N5Li24Li17H28O13H42(HeHe6CO11Li)35(He3O27HO5N21H49O39CH37B3)8(O41He27He46He22He17)12"], // B1354Be415C2012H826He1777Li1038N2045O1818
];

include_once dirname(__DIR__) . DIRECTORY_SEPARATOR ."xhprof.php";
$xhprof = new Xhprof();
foreach ($datas as $data){
    var_dump($data);

    $obj = new Solution();
    $result = $obj->countOfAtoms(...$data);
    // $result = $obj->($xhprof->tree($data));
    // $result = $obj->($xhprof->listNode($data));
    // $result = $obj->find(...$data);
    var_dump($result);
    // if(count($result)<=20){
    //     var_dump($result);
    // } else {
    //     var_dump(count($result));
    // }
    echo str_repeat(PHP_EOL, 3);
    echo str_repeat("<br>", 3);
}

// foreach ($datas as $data){
//     $obj = new $data[0][0](...$data[1][0]);
//
//     for ($i=1; $i<count($data[0]); $i++){
//         $func = $data[0][$i];
//         $param = $data[1][$i];
//         $result = $obj->$func(...$param);
//
//         echo $data[0][$i] ."(\"". implode(",", $data[1][$i]) ."\") ";
//             if($result === true){
//                 echo "True";
//             } elseif($result === false) {
//                 echo "False";
//             } elseif($result === null) {
//                 echo "Null";
//             } elseif(is_array($result)) {
//                 var_dump($result);
//             } else {
//                 echo $result;
//             }
//         echo PHP_EOL;
//     }
//
//     echo str_repeat(PHP_EOL, 3);
// }
$xhprof->end();
